package com.sunjob.common.util;

import org.springframework.data.domain.Page;

import java.util.*;

/**
 * Created by IntelliJ IDEA.
 *
 * @Author : Wangzurong
 * @create 2022-08-05 17:02
 * 分页工具类
 */
public class PageHelper<T>{
    public PageHelper(List<T> data){
        this.data = data;
    }
    private List<T> data;
    /* startPage起始页 pageSize 当前页记录条数 pageNum页数 */
    public List<List<T>> getPageInfo(int startPage,int pageSize,int pageNum){
        List<List<T>> rt = new LinkedList<>();
        int temp = pageSize*(startPage-1);
        flag:for(int j=1;j<=pageNum;j++){
            if(temp>=data.size()){
                break;
            }
            ArrayList<T> ts = new ArrayList<>();
            for(int t=0;t<pageSize && temp<data.size();t++){
                ts.add(data.get(temp++));
            }
            rt.add(ts);
        }
        return rt;
    }
    public List<T> getSinglePageInfo(int startPage,int pageSize){
        List<T> rt = new LinkedList<>();
        int temp = pageSize*(startPage-1);
        for(int t=0;t<pageSize && temp<data.size();t++){
            rt.add(data.get(temp++));
        }
        return rt;
    }
}

